This PR is a reworked and improved version of pyscript_autocomplete.
Better docstrings and more accurate type definitions for the Pyscript runtime and for Home Assistant domains, services, and entities with attributes.

How it works:

• Entity collection: pyscript.generate_stubs reads the HA entity registry, skips disabled entries, and records each entity along with its attributes. Attributes with names that aren’t valid Python identifiers are listed and excluded from the final code.
• Entity class generation: For each domain, a _domain_state class (subclass of StateVal) is created with the discovered attributes. Then a domain class is built where each entity is declared as a field of that state class.
• Service collection: Using async_get_all_descriptions, all HA service definitions are extracted. The generator converts selectors into python types, handles required/default fields and descriptions, and generates functions both at the domain level (as @staticmethod) and on the entity state class (if the service supports a target).
• AST-based code generation: Classes, declarations, and functions are constructed using the Python AST to ensure valid syntax, then converted to source with ast.unparse.
• File writing: The generator copies the bundled pyscript_builtins.py and writes the generated module to /pyscript/modules/stubs/.
• Ignored imports: from stubs import and from stubs.* import statements are ignored at runtime. Theoretically this could be a breaking change, but in practice, no one keeps real code in stubs.

Example for time domain:

class _time_state(StateVal):

    def set_value(self, time: str):
        """Sets the time.

        Args:
            time: The time to set. Example: 22:15"""
        ...

class time:
    time_settings_test: _time_state

    @staticmethod
    def set_value(*, entity_id: str, time: str):
        """Sets the time.

        Args:
            entity_id: Entity ID
            time: The time to set. Example: 22:15"""
        ...

In my HA installation, the generator produced about 5,900 lines of code, and it works perfectly.
Tested with PyCharm, VS Code, and Studio Code Server.